import { to as tos } from 'await-to-js';
import { Router } from 'vue-router';
import NProgress from 'nprogress';
import { getToken } from '@/utils/auth';
import { isHttp } from '@/utils/validate';
import { isRelogin } from '@/service/request';
import useUserStore from '@/store/modules/user';
import useSettingsStore from '@/store/modules/settings';
import usePermissionStore from '@/store/modules/permission';
// import { ElMessage } from 'element-plus';

const whiteList = ['/login', '/register'];

function setPermission(router: Router): void {

  router.beforeEach(async (to, /*from*/_, next) => {
    NProgress.start();
    // 已经有token，表明登录过
    // FIXME: token一旦设置之后就不会过期，每次切换之前应该检查token有效性
    if (getToken()) {
      to.meta.title && useSettingsStore().setTitle(to.meta.title as string);
      // 已经登录的情况下，再跳转到登录页则直接进主页
      if (to.path === '/login') {
        next({ path: '/' });
        NProgress.done();

      // 其他页面
      } else {
        const userStore = useUserStore();

        if (userStore.roleKeys.length === 0) {
          // 重新获取用户信息
          isRelogin.show = true;
          const [err] = await tos(userStore.getUserInfo());
          // 获取失败
          if (err) {
            ElMessage.error(err);
            await userStore.logout();
            next({ path: '/' });

          // 获取成功
          } else {
            isRelogin.show = false;

            const accessRoutes = await usePermissionStore().generateRoutes();

            // 根据roles权限生成可访问的路由表
            accessRoutes.forEach((route) => {
              if (!isHttp(route.path)) {
                // 动态添加可访问路由表
                router.addRoute(route);
              }
            });
            // 刷新当前页面，确保能够正确使用动态添加的路由
            next({ ...to, replace: true });
          }
        } else {
          next();
        }
      }
    } else {
      // 没有token
      if (whiteList.indexOf(to.path) !== -1) {
        // 在免登录白名单，直接进入
        next();
      } else {
        next(`/login?redirect=${to.fullPath}`); // 否则全部重定向到登录页
        NProgress.done();
      }
    }
  });

  router.afterEach(() => {
    NProgress.done();
  });
}

export default setPermission;
